iT邦幫忙

2018 iT 邦幫忙鐵人賽
DAY 15
0

透過上篇的創建模組,我們得到了許多的檔案,接下來我們需要透過檢查點的變數,將他轉換成常數節點,同時讓這些節點與圖像結合成獨立的檔案,這樣之後可以再APP端可以更簡單的使用這個模組檔

import tensorflow as tf
from tensorflow.python.tools import freeze_graph

input_graph = 'android_tf.pbtxt'
input_saver = ""
input_binary = False
checkpoint = 'android_tf.ckpt'
output_name = "O"
restore_name = "save/restore_all"
tensor_name = "save/Const:0"
output_frozen_graph = 'frozen_android_tf.pb'
clear_devices = True

freeze_graph.freeze_graph(input_graph,input_saver,
                          input_binary,checkpoint,output_name,
                          restore_name,tensor_name,
                          output_frozen_graph,clear_devices,"")

上一篇
[Day 14] 模組的創建及儲存
下一篇
[Day 16] 模組的優化
系列文
如何使用tensorflow在android上30
圖片
  直播研討會
圖片
{{ item.channelVendor }} {{ item.webinarstarted }} |
{{ formatDate(item.duration) }}
直播中

尚未有邦友留言

立即登入留言